package ru.emms.codemeter.tables;


import javax.swing.*;
import javax.swing.plaf.basic.BasicTableHeaderUI;
import javax.swing.table.DefaultTableCellRenderer;
import javax.swing.table.TableCellRenderer;
import javax.swing.table.TableColumn;
import javax.swing.table.TableColumnModel;
import java.awt.*;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.Iterator;

/**
 * Created by IntelliJ IDEA.
 * User: nastya.kuz
 * Date: 29.07.2010
 * Time: 17:35:20
 *
 * Модифицирование стандартного класса javax.swing.plaf.basic.BasicTableHeaderUI
 * с учетом того, что теперь работа происходит со сложным многоуровневым заголовком.
 */
public class GroupableTableHeaderUI
    extends BasicTableHeaderUI {

    /**
     * @param graphics  the Graphics context in which to paint
     * @param component the component being painted; this argument is often ignored, but might be used
     *                  if the UI object is stateless and shared by multiple components
     */
    public void paint(Graphics graphics, JComponent component) {
        Rectangle clipBounds = graphics.getClipBounds();
        if (header.getColumnModel() == null) {
            return;
        }
        ((GroupableTableHeader) header).setColumnMargin();
        int columnCount = 0;
        Dimension size = header.getSize();
        Rectangle cellRectangle = new Rectangle(0, 0, size.width, size.height);
        Hashtable<ColumnGroup, Rectangle> hashTable = new Hashtable<ColumnGroup, Rectangle>();
        Enumeration<TableColumn> enumeration = header.getColumnModel().getColumns();

        while (enumeration.hasMoreElements()) {
            cellRectangle.height = size.height;
            cellRectangle.y = 0;
            TableColumn aColumn = enumeration.nextElement();
            Iterator columnGroups = ((GroupableTableHeader) header).getColumnGroups(aColumn);
            if (columnGroups != null) {
                int groupHeight = 0;
                while (columnGroups.hasNext()) {
                    ColumnGroup columnGroup = (ColumnGroup) columnGroups.next();
                    Rectangle groupRectangle = hashTable.get(columnGroup);
                    if (groupRectangle == null) {
                        groupRectangle = new Rectangle(cellRectangle);
                        Dimension d = columnGroup.getSize(header.getTable());
                        groupRectangle.width = d.width - 5;
                        groupRectangle.height = d.height;
                        hashTable.put(columnGroup, groupRectangle);
                    }
                    paintCell(graphics, groupRectangle, columnGroup);
                    groupHeight += groupRectangle.height;
                    cellRectangle.height = size.height - groupHeight;
                    cellRectangle.y = groupHeight;
                }
            }
            cellRectangle.width = aColumn.getWidth();
            if (cellRectangle.intersects(clipBounds)) {
                paintCell(graphics, cellRectangle, columnCount);
            }
            cellRectangle.x += cellRectangle.width;
            columnCount++;
        }
    }

    /**
     * отрисовывает столбец в таблице
     *
     * @param graphics    графический контекст в котором происходит прорисовка
     * @param cellRect    размеры столбца
     * @param columnIndex индекс столбца
     */
    private void paintCell(Graphics graphics, Rectangle cellRect, int columnIndex) {
        TableColumn aColumn = header.getColumnModel().getColumn(columnIndex);
        TableCellRenderer renderer = new DefaultTableCellRenderer() {
            public Component getTableCellRendererComponent(JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
                JLabel header = new JLabel();
                header.setForeground(table.getTableHeader().getForeground());
                header.setBackground(table.getTableHeader().getBackground());
                header.setFont(table.getTableHeader().getFont());
                header.setHorizontalAlignment(JLabel.CENTER);
                header.setText(value.toString());
                header.setBorder(UIManager.getBorder("TableHeader.cellBorder"));
                return header;
            }
        };
        Component component = renderer.getTableCellRendererComponent(
            header.getTable(), aColumn.getHeaderValue(), false, false, -1, columnIndex);
        component.setBackground(UIManager.getColor("control"));
        rendererPane.add(component);
        rendererPane.paintComponent(graphics, component, header, cellRect.x, cellRect.y,
            cellRect.width, cellRect.height, true);
    }

    /**
     * отрисовывает группу столбцов в таблице
     *
     * @param graphics графический контекст в котором происходит прорисовка
     * @param cellRect размеры группы
     * @param cGroup   индекс столбца
     */
    private void paintCell(Graphics graphics, Rectangle cellRect, ColumnGroup cGroup) {
        TableCellRenderer renderer = cGroup.getHeaderRenderer();
        Component component = renderer.getTableCellRendererComponent(
            header.getTable(), cGroup.getHeaderValue(), false, false, -1, -1);
        rendererPane.add(component);
        rendererPane.paintComponent(graphics, component, header, cellRect.x, cellRect.y,
            cellRect.width, cellRect.height, true);
    }

    /**
     * вычисляет (суммарную по столбцам?) высоту header
     *
     * @return высота
     */
    private int getHeaderHeight() {
        int height = 0;
        TableColumnModel columnModel = header.getColumnModel();
        for (int column = 0; column < columnModel.getColumnCount(); column++) {
            TableColumn aColumn = columnModel.getColumn(column);
            TableCellRenderer renderer = aColumn.getHeaderRenderer();
            if (renderer == null) {
                // высота по умолчанию
                return 60;
            }
            Component component = renderer.getTableCellRendererComponent(
                header.getTable(), aColumn.getHeaderValue(), false, false, -1, column);
            int columnHeight = component.getPreferredSize().height;
            Iterator iterator = ((ArrayList) ((GroupableTableHeader) header).getColumnGroups(aColumn)).iterator();
            if (iterator != null) {
                while (iterator.hasNext()) {
                    ColumnGroup cGroup = (ColumnGroup) iterator.next();
                    columnHeight += cGroup.getSize(header.getTable()).height;
                }
            }
            height = Math.max(height, columnHeight);
        }
        return height;
    }

    /**
     * вычисляет размеры header исходя из предпочтительной ширины
     *
     * @param width ширина
     * @return размеры (ширина, высота)
     */
    private Dimension createHeaderSize(long width) {
        TableColumnModel columnModel = header.getColumnModel();
        width += columnModel.getColumnMargin() * columnModel.getColumnCount();
        width = Math.min(width, Integer.MAX_VALUE);
        return new Dimension((int) width, getHeaderHeight());
    }

    /**
     * возвращает предпочтительный размер для header
     *
     * @param component компонент, для которого вычисляется размер (в функции не используется)
     * @return возвращает предпочтительный размер компонента
     */
    public Dimension getPreferredSize(JComponent component) {
        long width = 0;
        Enumeration columns = header.getColumnModel().getColumns();
        while (columns.hasMoreElements()) {
            TableColumn aColumn = (TableColumn) columns.nextElement();
            width = width + aColumn.getPreferredWidth();
        }
        return createHeaderSize(width);
    }
}
